import cv2

def load_img(save_path,mode='gray'):
    if mode == 'gray':
        img = cv2.imread(save_path,0).astype('float')
    else:
        img = cv2.imread(save_path).astype('float')
    return img/255
    # numpy [0,1]